package com.atguigu.algorithm.recursion;

import java.util.Arrays;

/**
 * 八皇后问题-使用递归回溯法
 *
 * @author: zhaozongyi
 * @date: 2022-08-22 11:52
 */
public class Queen8 {

    //定义数组, 保存皇后放置位置的结果,比如 arr = {0 , 4, 7, 5, 2, 6, 1, 3}
    static int[] queen = new int[8];
    static int count = 0;
    static int judgeCount = 0;

    public static void main(String[] args) {
        solution(0);
        System.out.printf("一共有%d解法", count);
        System.out.printf("一共判断冲突的次数%d次", judgeCount); // 1.5w
    }

    /**
     * 递归回溯
     * 编写一个方法，放置第n个皇后
     * 特别注意： solution 是 每一次递归时，进入到solution中都有  for(int i = 0; i < max; i++)，因此会有回溯
     * @param n
     */
    public static void solution(int n) {
        if (n == 8) {
            print();
        } else {
            //依次放入皇后，并判断是否冲突
            for (int i = 0; i < 8; i++) {
                //先把当前这个皇后 n , 放到该行的第1列
                queen[n] = i;
                //判断当放置第n个皇后到i列时，是否冲突
                if(judge(n)) { // 不冲突
                    //接着放n+1个皇后,即开始递归
                    solution(n+1); //
                }
                //如果冲突，就继续执行 array[n] = i; 即将第n个皇后，放置在本行得 后移的一个位置
            }
        }
    }

    /**
     * 判断位置是否冲突
     * 1. array[i] == array[n]  表示判断 第n个皇后是否和前面的n-1个皇后在同一列
     * 2. Math.abs(n-i) == Math.abs(array[n] - array[i]) 表示判断第n个皇后是否和第i皇后是否在同一斜线
     * 3. 判断是否在同一行, 没有必要，n 每次都在递增
     * @return
     */
    public static boolean judge (int n) {
        judgeCount++;
        for(int i = 0; i < n; i++) {
            if(queen[i] == queen[n] || Math.abs(n-i) == Math.abs(queen[n] - queen[i]) ) {
                return false;
            }
        }
        return true;
    }

    /**
     * 打印位置
     */
    public static void print() {
        count++;
        System.out.println(Arrays.toString(queen));
    }

}
